import pandas as pd
import os


def GetMalwareList():
    path = r'./data/feature/malware/'
    files = os.listdir(path)
    for i in range(len(files)):
        files[i] = files[i].split('.apk')[0]
    save = pd.DataFrame({'sha256': files})
    save.to_csv('./data/feature/malware_sha256.csv', index=False, sep=',')

